import os
from dotenv import load_dotenv

# 加载 .env 文件
load_dotenv()

class Config:
    # 存储配置
    DATA_DIR = os.getenv('DATA_DIR', 'data')
    MAX_RECORDS = int(os.getenv('MAX_RECORDS', '1000'))

config = Config()

class DevelopmentConfig(Config):
    DEBUG = True
    
class ProductionConfig(Config):
    DEBUG = False
    
